// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

import { IExplanationDashboardData } from "@responsible-ai/core-ui";

export const irisData: IExplanationDashboardData = {
  dataSummary: {
    classNames: ["setosa", "versicolor", "virginica"],
    featureNames: [
      "sepal length (cm)",
      "sepal width (cm)",
      "petal length (cm)",
      "petal width (cm)"
    ]
  },
  modelInformation: { method: "classifier", modelClass: "blackbox" },
  precomputedExplanations: {
    globalFeatureImportance: {
      intercept: 1,
      scores: [
        0.017800017217248786, 0.011517443776666336, 0.32944454026478104,
        0.08439875569117994
      ]
    },
    localFeatureImportance: {
      intercept: [0.1, 0.25, 0.2],
      scores: [
        [
          [
            -0.0016556448742914842, -0.008024639554334245, -0.2683597779520689,
            -0.03179943364824328
          ],
          [
            -0.0018031235219435893, -0.02077440003136455, -0.27412647996582407,
            -0.01081608410825019
          ],
          [
            0.0017704720900889592, 0.03411859734385725, 0.5906992682680244,
            0.03225996862610736
          ],
          [
            -0.004411795870242291, -0.005052130378713493, -0.2881623105078052,
            -0.019889995420295792
          ],
          [
            0.005139042323232923, 0.013756118003964646, 0.5738510236632185,
            0.0443766114706553
          ],
          [
            -0.002704138711798587, 0.00022094189248925145, -0.28143996268943994,
            -0.033429280273780004
          ],
          [
            0.0038590296715981243, 0.01376134830738407, 0.6034797008591676,
            0.02939284471299619
          ],
          [
            -0.004615238612371453, -0.002020110801213726, -0.2907394220978406,
            -0.014368155047137177
          ],
          [
            -0.004867411769868851, -0.005463548923605177, -0.28815543966094725,
            -0.013234764042659386
          ],
          [
            -0.002197530515119811, -0.008545364708717057, -0.2771893861846843,
            -0.016163777510307953
          ],
          [
            -0.0016222959228014144, -0.008145800012010262, -0.2900618218796472,
            -0.011382684706985946
          ],
          [
            -0.003232624340801238, -0.0008185925951046114, -0.29179560999642334,
            -0.015585607163696391
          ],
          [
            -0.002009518623111828, -0.0052202487396008634, -0.29370179419361975,
            -0.011575747757903154
          ],
          [
            -0.003922041456244474, -0.0059054120204183325, -0.2860448840835026,
            -0.015396755089337244
          ],
          [
            -0.0020086575003105955, -0.004146534862599416, -0.29054643998630575,
            -0.013048240735613226
          ],
          [
            0.004571532600452843, 0.017900668233482908, 0.5907681312989816,
            0.0386638867021184
          ],
          [
            -0.0017578126265546667, -0.0046735441652921705, -0.2878794424833619,
            -0.01565228137010405
          ],
          [
            0.0002736181980413366, -0.008904939520222588, -0.28923647360504795,
            -0.013064959064932602
          ],
          [
            0.006018592008084578, -0.0008956494251255998, 0.5849378803512715,
            0.039842386605087654
          ],
          [
            0.002090329893318754, 0.02433097453232952, 0.6040104064278179,
            0.02450515630470418
          ],
          [
            -0.0006303901945865775, -0.007089648663357012, -0.27573968613203187,
            -0.020131243405170085
          ],
          [
            1.2513648310535919e-5, -0.003276527696037812, -0.2895047874607707,
            -0.015135647903689098
          ],
          [
            0.00995145614227283, 0.021699652427084326, 0.4864834489014331,
            0.0746891040568296
          ],
          [
            0.007939296262232554, -0.004198923042389557, 0.5847869230513024,
            0.04360605319316391
          ],
          [
            -0.0023077309625416376, -0.006360708436080914, -0.27807432731771164,
            -0.01656545713884483
          ],
          [
            0.003408685823244384, 0.01155377297086152, 0.6311723103128468,
            0.01974567766808777
          ],
          [
            0.007195726205551478, 0.04370714505210227, 0.49080027759598865,
            0.05835947535834396
          ],
          [
            -0.0021846579631739915, -0.0050905648989428565,
            -0.28894500176703874, -0.014314772542068033
          ],
          [
            0.005065197264792362, -0.047054333249719596, -0.18723910746925188,
            -0.00908910416218034
          ],
          [
            0.006190523472554277, 0.01631903116468536, 0.5538916150892038,
            0.04419893224189908
          ]
        ],
        [
          [
            -0.006819789435792563, 0.00273823901216141, -0.007120667498136888,
            -0.2801573698829855
          ],
          [
            -0.007204644229200419, -0.005921962430237904, 0.5668882304304849,
            0.1181593102991021
          ],
          [
            -0.01542668141516454, -0.007345138119860178, -0.38681025172146666,
            0.11755122690534564
          ],
          [
            0.04721210621906223, -4.486611729345724e-5, -0.299480854636605,
            -0.05030662351106863
          ],
          [
            -0.02765098377369235, -0.003325336154981262, -0.3540294330499134,
            0.10958179993531286
          ],
          [
            0.022406272213126993, 0.010598837681640425, -0.18247068340750203,
            -0.15544820122597647
          ],
          [
            -0.026761829927327813, -0.000832854321289489, -0.3752443169878133,
            0.11699532343984964
          ],
          [
            0.06203524744806993, 0.016087789144412834, 0.4048809568805294,
            0.05688284344463357
          ],
          [
            0.07794880882651428, -0.012534592399017691, 0.3463533563507202,
            0.08202795242419042
          ],
          [
            -0.0015957382850068647, 0.004745561738964343, 0.5879015573235302,
            0.07449260716093331
          ],
          [
            0.008522709815987087, -0.022612642849727596, -0.17251677793346729,
            -0.06378269502356357
          ],
          [
            0.024381344343351485, 0.02019203301586503, 0.5008184314893709,
            0.06238790310071396
          ],
          [
            0.007386259567608633, -0.00824790046706761, 0.4129810281333831,
            0.15144314410611392
          ],
          [
            0.04370037644509897, -0.009970146396922785, 0.4377093244848807,
            0.05704577459052673
          ],
          [
            0.011874035320573517, -0.0027887295865828, 0.39072781229158543,
            0.08264885813751682
          ],
          [
            -0.02782186018432961, -0.003592484604806906, -0.3754455446654346,
            0.11968956104596978
          ],
          [
            0.0008911118351712655, -0.0007453905344288192, 0.48190137648794173,
            0.059282559771658416
          ],
          [
            -0.03280014405330334, -0.01407166270698948, 0.519523422712701,
            0.13621317149083068
          ],
          [
            -0.03382827669815139, 0.00211441814183555, -0.3453022952742722,
            0.10746359007953737
          ],
          [
            -0.018168424388722848, -0.002215312531993613, -0.3830603055303358,
            0.11444921063491187
          ],
          [
            -0.03372245258537959, -0.00954897963721139, 0.07235759982710245,
            -0.23215699229270528
          ],
          [
            -0.029782750746733366, 0.006978539544805873, 0.47551256851917656,
            0.05619764029980939
          ],
          [
            -0.03695280997592551, -0.010924073899760217, -0.2739854388933698,
            0.08096307093898686
          ],
          [
            -0.042881337217902465, 0.0029023087015197235, -0.34247408632934284,
            0.11030198850552253
          ],
          [
            0.017238719655923995, -0.020823156370027365, 0.1997561650434761,
            -0.13117335850290068
          ],
          [
            -0.0322532438347031, 0.002984331790309075, -0.3992102445900268,
            0.1309779215733985
          ],
          [
            -0.029564806104662417, -0.0224045470085745, -0.27767591050902357,
            0.08298116899691427
          ],
          [
            0.002690676872684483, 0.003203134221022197, 0.5537603520423295,
            0.09530588432808129
          ],
          [
            -0.03373313569272543, 0.03600036843979926, 0.5077380389948398,
            0.08234376739518323
          ],
          [
            -0.03059080776697387, -0.004923499670592274, -0.3237450780873215,
            0.09669429552554165
          ]
        ],
        [
          [
            0.00847543431008399, 0.0052864005421727756, 0.2754804454502057,
            0.3119568035312289
          ],
          [
            0.009007767751143952, 0.026696362461602355, -0.2927617504646607,
            -0.10734322619085201
          ],
          [
            0.01365620932507558, -0.026773459223997084, -0.20388901654655772,
            -0.14981119553145295
          ],
          [
            -0.042800310348819925, 0.0050969964960069225, 0.5876431651444104,
            0.07019661893136453
          ],
          [
            0.02251194145045958, -0.01043078184898344, -0.21982159061330514,
            -0.15395841140596805
          ],
          [
            -0.019702133501328378, -0.010819779574129745, 0.4639106460969421,
            0.18887748149975636
          ],
          [
            0.02290280025572973, -0.012928493986094608, -0.2282353838713544,
            -0.14638816815284572
          ],
          [
            -0.05742000883569845, -0.01406767834319908, -0.11414153478268865,
            -0.04251468839749631
          ],
          [
            -0.07308139705664538, 0.017998141322622993, -0.05819791668977295,
            -0.06879318838153094
          ],
          [
            0.0037932688001265646, 0.00379980296975288, -0.3107121711388457,
            -0.058328829650625524
          ],
          [
            -0.006900413893185686, 0.030758442861737956, 0.4625785998131144,
            0.07516537973054965
          ],
          [
            -0.0211487200025504, -0.019373440420760363, -0.2090228214929476,
            -0.0468022959370174
          ],
          [
            -0.005376740944496694, 0.01346814920666848, -0.11927923393976327,
            -0.13986739634821074
          ],
          [
            -0.039778334988854525, 0.015875558417341215, -0.15166444040137803,
            -0.04164901950118921
          ],
          [
            -0.009865377820262942, 0.0069352644491821325, -0.10018137230527968,
            -0.06960061740190351
          ],
          [
            0.023250327583876726, -0.01430818362867603, -0.21532258663354714,
            -0.15835344774808813
          ],
          [
            0.0008667007913833458, 0.0054189346997211285, -0.1940219340045799,
            -0.04363027840155406
          ],
          [
            0.03252652585526207, 0.02297660222721204, -0.23028694910765307,
            -0.12314821242589796
          ],
          [
            0.027809684690066797, -0.0012187687167099087, -0.23963558507699934,
            -0.14730597668462475
          ],
          [
            0.016078094495404052, -0.022115662000335823, -0.2209501008974819,
            -0.1389543669396161
          ],
          [
            0.03435284277996628, 0.01663862830056849, 0.20338208630492938,
            0.25228823569787534
          ],
          [
            0.02977023709842288, -0.0037020118487679776, -0.18600778105840593,
            -0.04106199239612035
          ],
          [
            0.02700135383365268, -0.010775578527324164, -0.21249801000806318,
            -0.15565217499581663
          ],
          [
            0.03494204095567005, 0.0012966143408698755, -0.24231283672195955,
            -0.1539080416986866
          ],
          [
            -0.014930988693382278, 0.027183864806108446, 0.07831816227423555,
            0.14773881564174546
          ],
          [
            0.028844558011458687, -0.014538104761170484, -0.23196206572282002,
            -0.15072359924148618
          ],
          [
            0.022369079899110925, -0.021302598043527715, -0.2131243670869651,
            -0.14134064435525812
          ],
          [
            -0.0005060189095106582, 0.0018874306779206873, -0.2648153502752907,
            -0.08099111178601326
          ],
          [
            0.028667938427933165, 0.011053964809920325, -0.3204989315255878,
            -0.07325466323300289
          ],
          [
            0.024400284294419552, -0.01139553149409299, -0.2301465370018823,
            -0.14089322776744082
          ]
        ]
      ]
    }
  },
  predictedY: [
    2, 1, 0, 2, 0, 2, 0, 1, 1, 1, 2, 1, 1, 1, 1, 0, 1, 1, 0, 0, 2, 1, 0, 0, 2,
    0, 0, 1, 1, 0
  ],
  probabilityY: [
    [0.010884802029477083, 0.014159913639079359, 0.9749552843314434],
    [0.0132042104310326, 0.9774404355139816, 0.009355354054985684],
    [0.979572604386493, 0.01348865709268723, 0.006938738520819898],
    [0.003208065881358132, 0.002899263397928044, 0.993892670720714],
    [0.9578470935194863, 0.0300955484005588, 0.012057358079955005],
    [0.00337185827588573, 0.0006057267051219008, 0.9960224150189924],
    [0.9712172216095609, 0.01967582364725199, 0.00910695474318709],
    [0.008981371499852037, 0.8454063383614786, 0.14561229013866955],
    [0.00900313366133425, 0.7993150266462401, 0.19168183969242578],
    [0.016628239139585853, 0.971063489382254, 0.012308271478160247],
    [0.009511695536970148, 0.055130095453061585, 0.9353582090099684],
    [0.009291863962389358, 0.9132992133931344, 0.07740892264447628],
    [0.008216988744179358, 0.869082032783871, 0.12270097847194981],
    [0.009455205408912317, 0.8340048305674166, 0.15653996402367149],
    [0.010974424973585953, 0.7879814776069259, 0.20104409741948803],
    [0.9726285168934508, 0.018349173035231564, 0.009022310071317496],
    [0.010761217413102162, 0.8468491590041755, 0.14238962358272256],
    [0.009791544066253153, 0.9143842888870718, 0.07582416704667509],
    [0.9506275075977331, 0.03596693769278224, 0.013405554709484837],
    [0.9756611652165853, 0.016524669627692574, 0.007814165155722254],
    [0.01713332966326939, 0.10244867675563911, 0.8804179935810915],
    [0.012819848646227922, 0.8144254990608915, 0.17275465229288067],
    [0.9135479595860349, 0.06462024961376427, 0.02183179080020075],
    [0.9528576475227243, 0.03336837510362991, 0.013773977373645852],
    [0.01741607420323595, 0.370517871270305, 0.6120660545264592],
    [0.9866047448334555, 0.008018266382810606, 0.00537698878373407],
    [0.9207869222704013, 0.05885540681848671, 0.020357670911112027],
    [0.010189300887191319, 0.9604795489079505, 0.029331150204858096],
    [0.08240695044205551, 0.8978685405809298, 0.019724508977014808],
    [0.9413244000267575, 0.04295441144448689, 0.01572118852875551]
  ],
  testData: [
    [5.8, 2.8, 5.1, 2.4],
    [6, 2.2, 4, 1],
    [5.5, 4.2, 1.4, 0.2],
    [7.3, 2.9, 6.3, 1.8],
    [5, 3.4, 1.5, 0.2],
    [6.3, 3.3, 6, 2.5],
    [5, 3.5, 1.3, 0.3],
    [6.7, 3.1, 4.7, 1.5],
    [6.8, 2.8, 4.8, 1.4],
    [6.1, 2.8, 4, 1.3],
    [6.1, 2.6, 5.6, 1.4],
    [6.4, 3.2, 4.5, 1.5],
    [6.1, 2.8, 4.7, 1.2],
    [6.5, 2.8, 4.6, 1.5],
    [6.1, 2.9, 4.7, 1.4],
    [4.9, 3.6, 1.4, 0.1],
    [6, 2.9, 4.5, 1.5],
    [5.5, 2.6, 4.4, 1.2],
    [4.8, 3, 1.4, 0.3],
    [5.4, 3.9, 1.3, 0.4],
    [5.6, 2.8, 4.9, 2],
    [5.6, 3, 4.5, 1.5],
    [4.8, 3.4, 1.9, 0.2],
    [4.4, 2.9, 1.4, 0.2],
    [6.2, 2.8, 4.8, 1.8],
    [4.6, 3.6, 1, 0.2],
    [5.1, 3.8, 1.9, 0.4],
    [6.2, 2.9, 4.3, 1.3],
    [5, 2.3, 3.3, 1],
    [5, 3.4, 1.6, 0.4]
  ]
};
